Random Attention: Rethinking KV Cache Eviction for Efficient Reasoning
随机注意力:重新思考 KV 缓存淘汰以实现高效推理
🧭 LLM 推理效率 · KV 缓存随机淘汰
Salesforce AI Research · University of Illinois Urbana-Champaign
提出 Random Attention,保留 prompt 并在各注意力头内随机淘汰 KV 缓存,匹配最强选择器同时吞吐提升 32-43%
📌 工作参考价值(4 点)
- 在部署长上下文 LLM 服务(如电商 Agent 的多轮对话或长链推理)时,可将 KV cache 淘汰策略简化为“保留 prompt + 每头随机淘汰”,省去 token 重要性打分,降低额外计算和延迟,实现 32-43% 的推理吞吐提升。
- 显式保护 prompt 是稳定性的关键:随机淘汰由于天然保留 prompt 而表现出与复杂选择器相当的性能,提示我们在设计缓存管理策略时应优先确保 prompt 完整,而非依赖复杂重要性估计。
- 推理轨迹在文本和注意力头两个层面存在冗余,因此随机抽样足以保留必要信息,这对长 CoT 生成场景尤其有借鉴:可放宽对精确 token 选择的假设,采用更轻量的随机策略。
- 可作为强 baseline:在电商搜索推荐系统的 LLM 模块中,如果采用 KV 压缩,先测试随机淘汰,它实现简单、无需训练或计算分数,且能匹配最优方法。